#include <stdio.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/time.h>
#include <sys/select.h>


#define BUF_SIZE 1024

void error_handling(char *buf);
void read_routine(int sock , char *buf);
void write_routine(int sock, char *buf);



int main(int argc, char *argv[]){
	
	int sock;
	pid_t pid;
	char buf[BUF_SIZE];
	int str_len;
	struct sockaddr_in serv_adr;

	if (argc!=3)
	{
		printf("usage error\n");
		exit(1);
	}
	sock = socket(PF_INET, SOCK_STREAM, 0);
	if (sock==-1)
	{
		error_handling("socket create error");
	}

	memset(&serv_adr, 0,sizeof(serv_adr));
	serv_adr.sin_family=AF_INET;
	serv_adr.sin_addr.s_addr=inet_addr(argv[1]);
	serv_adr.sin_port=htons(atoi(argv[2]));

	if (connect(sock,(struct sockaddr*)&serv_adr, sizeof(serv_adr))==-1)
	{
		error_handling("connect error");
	}else{
		puts("connected......");
	}
	
	pid = fork();
	if (pid == 0)
		write_routine(sock,buf);
	else
		read_routine(sock,buf);
	close(sock);

	
	return 0;

}


void read_routine(int sock , char *buf){
	while (1)
	{
		int str_len =read(sock, buf, BUF_SIZE);
		if(str_len ==0)
			return;

		buf[str_len]=0;
		printf("message from server: %s", buf);
	}
}
void write_routine(int sock, char *buf){

	while(1){
		fgets(buf,BUF_SIZE, stdin);
		if (!strcmp(buf, "q\n")|| !strcmp(buf, "Q\n"))
		{
			shutdown(sock,SHUT_WR);
			return;
		}
		write(sock,buf,strlen(buf));
	}
	
}

void error_handling(char* buf){
	fputs(buf, stderr);
	fputs('\n', stderr);
	exit(1);
}